package socket;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

/**
 *
 */
public class Server5 {
    private ServerSocket serverSocket;

    //V4：该集合用于保存所有客户端的输出流，用于广播消息
    private List<PrintWriter> allOut = new ArrayList<>();

    public Server5() {
        try {
            /*
            如果端口被其他程序占用了在，则会发生异常：
            解决办法：
            1.更换端口号
            2.杀死占用该端口的进程（通常发生在服务端启动两次导致的情况下）
             */
            System.out.println("正在启动服务端");
            serverSocket = new ServerSocket(8088);
            System.out.println("服务端启动完毕！");
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    public void start() {
        /*
        ServerSocket提供的主要方法：
        Socket accept（）
        用于接收客户端的连接，并返回一个Socket实例与连接的客户端进行交互
        该方法是一个阻塞方法，调用后程序会“卡住”，直到一个客户端连接为止
         */
        try {
            while (true) {
                System.out.println("等待客户端连接");
                Socket socket = serverSocket.accept();
                System.out.println("一个客户端连接上了");
                //启动一个线程负责与该客户端交互
                ClientHandler handler = new Server5.ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();
            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    public static void main(String[] args) {
        Server5 server = new Server5();
        server.start();
    }

    /**
     * 该线程任务负责与指定的客户端交互
     */
    private class ClientHandler implements Runnable {
        //传参socket技巧：定义个全局变量对应参数socket，在构造函数里传参
        private Socket socket;
        private String host;//记录该客户端的IP地址

        public ClientHandler(Socket socket) {
            this.socket = socket;
            host = socket.getInetAddress().getHostAddress();
        }

        public void run() {
            //V5:提高pw的作用域，以便在finally中可以将其从allOut集合中删除
            PrintWriter pw = null;
            try {
                InputStream in = socket.getInputStream();//低级流
                InputStreamReader isr = new InputStreamReader(in, StandardCharsets.UTF_8);
                BufferedReader br = new BufferedReader(isr);
                //当我们通过br.readline()读取来自远端计算机（客户端）发送过来的字符串时，
                //远端计算机只有调用socket.close()正常与我们断开连接时，readLine()才会返回null。

                //V3:通过socket获取输出流用于给客户端发送消息
                OutputStream out = socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out, StandardCharsets.UTF_8);
                BufferedWriter bw = new BufferedWriter(osw);
                pw = new PrintWriter(bw, true);

                //V4：将对应该客户端的输出流存入到共享台allOut中，便于其它线程可以访问到
                allOut.add(pw);
                //V5: 广播消息，该客户端上线了，当前在线人数：XX
                sendMessage(host+"上线了,当前在线人数："+allOut.size());

                String message;
                while ((message = br.readLine()) != null) {
                    //V5：读取客户端消息广播时改为调用sendMessage方法
                    sendMessage(host+"："+message);
                }
            } catch (IOException e) {
                throw new RuntimeException(e);
            }//V5：在finally中处理客户端断开后的操作
            finally {
                //将该客户端的输出流从allOut集合中删除
                allOut.remove(pw);
                //广播该客户端下线了
                sendMessage(host+"下线了,当前在线人数："+allOut.size());
            }
        }

        /**
         * V5新功能：实现广播消息方法：
         * 将消息转发给全部客户端
         * @param message
         */
        private void sendMessage(String message) {
            System.out.println( message);
            //V4：遍历allOut集合，给所有客户端发送这个消息
            for (PrintWriter o : allOut) {
                o.println( message);
            }
        }
    }
}
